package cn.iocoder.yudao.module.ai.service.label;

import cn.iocoder.yudao.framework.exception.BusinessException;
import cn.iocoder.yudao.framework.security.core.util.SecurityFrameworkUtils;
import cn.iocoder.yudao.module.ai.dal.dataobject.label.LabelGroupDO;
import cn.iocoder.yudao.module.system.api.dept.DeptApi;
import org.apache.commons.collections4.CollectionUtils;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import cn.iocoder.yudao.module.ai.controller.admin.label.vo.*;
import cn.iocoder.yudao.module.ai.dal.dataobject.label.LabelDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;

import cn.iocoder.yudao.module.ai.dal.mysql.label.LabelMapper;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.ai.enums.ErrorCodeConstants.*;

/**
 * 标签管理 Service 实现类
 *
 * @author 侯俊青
 */
@Service
@Validated
public class LabelServiceImpl implements LabelService {

    @Resource
    private LabelMapper labelMapper;
    @Resource
    private DeptApi deptApi ;

    @Override
    public Long createLabel(LabelSaveReqVO createReqVO) {
        // 插入
        LabelDO label = BeanUtils.toBean(createReqVO, LabelDO.class);
        label.setId(null);
        label.setSuplierId(deptApi.queryFirstDept(SecurityFrameworkUtils.getLoginUserDeptId()));
        labelMapper.insert(label);
        // 返回
        return label.getId();
    }

    @Override
    public void updateLabel(LabelSaveReqVO updateReqVO) {
        // 校验存在
        validateLabelExists(updateReqVO.getId());
        // 更新
        LabelDO updateObj = BeanUtils.toBean(updateReqVO, LabelDO.class);
        labelMapper.updateById(updateObj);
    }

    @Override
    public void deleteLabel(Long id) {
        // 校验存在
        validateLabelExists(id);
        // 查询有无关联组
        List<LabelGroupDO> labelGroupDOS = labelMapper.selectGroupInfoByLabelId(id);
        if(CollectionUtils.isNotEmpty(labelGroupDOS)){
            String name = labelGroupDOS.get(0).getName();
            throw new BusinessException("标签已经被【"+name+"】等"+labelGroupDOS.size()+"个标签组关联，无法删除");
        }
        // 删除
        labelMapper.deleteById(id);
    }

    private void validateLabelExists(Long id) {
        if (labelMapper.selectById(id) == null) {
            throw exception(LABEL_NOT_EXISTS);
        }
    }

    @Override
    public LabelDO getLabel(Long id) {
        return labelMapper.selectById(id);
    }

    @Override
    public PageResult<LabelDO> getLabelPage(LabelPageReqVO pageReqVO) {
        return labelMapper.selectPage(pageReqVO);
    }

}